function with multiple arguments in python